tensor([[[0.2900, 0.2900],
         [0.2900, 0.2900]],

        [[0.3100, 0.3100],
         [0.3100, 0.3100]],

        [[0.4000, 0.4000],
         [0.4000, 0.4000]]], grad_fn=<RoundBackward1>)